from pymongo import MongoClient
import datetime
from dotenv import load_dotenv
import os

# 加载环境变量
load_dotenv()

# 连接到MongoDB
mongo_uri = os.getenv('MONGO_URI', 'mongodb://localhost:27017/medical_system')
client = MongoClient(mongo_uri)
db = client.medical_system

# 获取所有集合
collections = db.list_collection_names()
print(f"数据库中的所有集合: {collections}")

# 检查医嘱集合是否存在
if 'medical_orders' not in collections:
    # 创建医嘱集合
    db.create_collection('medical_orders')
    print("已成功创建medical_orders集合")
    
    # 创建索引
    db.medical_orders.create_index('orderId', unique=True)
    print("已为medical_orders集合创建orderId索引")
else:
    print("medical_orders集合已存在")
    
    # 检查索引是否存在
    indexes = db.medical_orders.index_information()
    print(f"medical_orders集合的索引: {list(indexes.keys())}")
    
    if 'orderId_1' not in indexes:
        db.medical_orders.create_index('orderId', unique=True)
        print("已添加缺失的orderId索引")
    else:
        print("orderId索引已存在")

# 检查集合中是否有数据
order_count = db.medical_orders.count_documents({})
print(f"medical_orders集合中有 {order_count} 条记录")

# 关闭连接
client.close()

print("\n提示: 医嘱功能已准备就绪，可以开始使用！") 